/**
 * 该文件可自行根据业务逻辑进行调整
 */
import type { HttpResponse } from '@vben/request';

import { useAppConfig } from '@vben/hooks';
import { preferences } from '@vben/preferences';
import {
  authenticateResponseInterceptor,
  errorMessageResponseInterceptor,
  RequestClient,
} from '@vben/request';
import { useAccessStore, useUserStore } from '@vben/stores';

import { message } from 'ant-design-vue';

import { useAuthStore } from '#/store';

import { refreshTokenApi } from './core';

const { apiURL } = useAppConfig(import.meta.env, import.meta.env.PROD);

const forAiJwt = import.meta.env.VITE_APP_AI_JWT;

const aiUrl = 'https://ai.doulingyang.cn/api/';

function createRequestClient(baseURL: string) {
  const client = new RequestClient({
    baseURL,
    timeout: 1000 * 60 * 5,
  });

  /**
   * 重新认证逻辑
   */
  async function doReAuthenticate() {
    console.warn('Access token or refresh token is invalid or expired. ');
    const accessStore = useAccessStore();
    const authStore = useAuthStore();
    accessStore.setAccessToken(null);
    if (
      preferences.app.loginExpiredMode === 'modal' &&
      accessStore.isAccessChecked
    ) {
      accessStore.setLoginExpired(true);
    } else {
      await authStore.logout();
    }
  }

  /**
   * 刷新token逻辑
   */
  async function doRefreshToken() {
    const accessStore = useAccessStore();
    const resp = await refreshTokenApi({
      refreshToken: accessStore.refreshToken,
    });
    // console.log('resp', resp);
    const newToken = resp?.access_token; // || accessStore.accessToken; // formatToken(accessStore.accessToken);
    const newRFToken = resp?.refresh_token; // || accessStore.accessToken; // formatToken(accessStore.accessToken);
    // console.log('newToken', newToken);
    accessStore.setAccessToken(newToken);
    accessStore.setRefreshToken(newRFToken);
    return newToken;
  }

  function formatToken(token: null | string) {
    return token ? `Bearer ${token}` : null;
  }

  // 请求头处理
  client.addRequestInterceptor({
    fulfilled: async (config) => {
      const userStore = useUserStore();
      const accessStore = useAccessStore();
      config.headers.Authorization =
        config.baseURL === aiUrl
          ? forAiJwt
          : formatToken(accessStore.accessToken);
      config.headers['Accept-Language'] = preferences.app.locale;
      config.headers.userId = userStore?.userInfo?.userId || '';
      config.headers.system = '1';
      return config;
    },
  });

  // response数据解构
  client.addResponseInterceptor<HttpResponse>({
    fulfilled: (response) => {
      const { data: responseData, status } = response;

      const { code, data, msg } = responseData;
      if (status === 200 && status < 400 && (code === 200 || code === 0)) {
        // console.log('data', data);
        return data || {};
      }

      if (code && code !== 200) {
        message.error(msg);
        client.addRequestInterceptor({
          fulfilled: async (config) => {
            const accessStore = useAccessStore();
            config.headers.Authorization = formatToken(accessStore.accessToken);
            config.headers['Accept-Language'] = preferences.app.locale;
            config.headers.system = '1';
            return config;
          },
        });
        return false;
      }

      throw Object.assign({}, response, { response });
    },
  });

  // token过期的处理
  client.addResponseInterceptor(
    authenticateResponseInterceptor({
      client,
      doReAuthenticate,
      doRefreshToken,
      enableRefreshToken: preferences.app.enableRefreshToken,
      formatToken,
    }),
  );

  // 通用的错误处理,如果没有进入上面的错误处理逻辑，就会进入这里
  client.addResponseInterceptor(
    errorMessageResponseInterceptor((msg: string, error) => {
      // 这里可以根据业务进行定制,你可以拿到 error 内的信息进行定制化处理，根据不同的 code 做不同的提示，而不是直接使用 message.error 提示 msg
      // 当前mock接口返回的错误字段是 error 或者 message
      const responseData = error?.response?.data ?? {};
      const errorMessage = responseData?.error ?? responseData?.message ?? '';
      // 如果没有错误信息，则会根据状态码进行提示
      message.error(errorMessage || msg);
    }),
  );

  return client;
}

export const requestClient = createRequestClient(apiURL);
export const requestClientDon = createRequestClient(aiUrl);

export const baseRequestClient = new RequestClient({ baseURL: apiURL });
